package com.example.demo.controller;

import org.springframework.web.bind.annotation.CrossOrigin;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

import java.io.File;
import java.io.FileInputStream;
import java.util.ArrayList;

@RestController
@RequestMapping("/api")
@CrossOrigin(origins = "http://localhost:5173")
public class PictureController {

    @GetMapping("/getPicture")
    public ArrayList<Object> getPi() {
        ArrayList list = new ArrayList();
        // 获取文件夹路径
        String filePath = "C:\\Users\\WoTing\\Desktop\\test";
        File fileFolder = new File(filePath);
        // 判断文件夹是否存在
        if (fileFolder.exists() && fileFolder.isDirectory()) {
            System.out.println("-----------文件夹存在");
            // 遍历文件
            File[] files = fileFolder.listFiles();
            for (File file : files) {
                boolean flag = isPicture(file);
                if (flag) {
                    String picPath = "file:\\\\" + filePath + "\\" + file.getName();
                    list.add(picPath);
                }
            }
            // 判断是否为图片
        } else {
            System.out.println("-----------指定的文件夹路径不存在或不是一个文件夹。");
        }
        return list;
    }


    /**
     *
     */
    public boolean isPicture(File file) {
        String fileName = file.getName();
        // 111111.jpg
        // 检查文件的后缀名是否为图片格式（可根据需要自定义）
        String extension = fileName.substring(fileName.lastIndexOf(".") + 1).toLowerCase();
        return extension.equals("jpg") || extension.equals("jpeg") || extension.equals("png");
    }

}
